tensor([[[0.2800, 0.2800],
         [0.2800, 0.2800]],

        [[0.4700, 0.4700],
         [0.4700, 0.4700]],

        [[0.2500, 0.2500],
         [0.2500, 0.2500]]], grad_fn=<RoundBackward1>)